package com.zhn;

import java.util.HashSet;
import java.util.Set;

//编写一个算法来判断一个数 n 是不是快乐数。
//
//「快乐数」 定义为：
//
//对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
//然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
//如果这个过程 结果为 1，那么这个数就是快乐数。
//如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
//

public class IsHappy {
    public boolean isHappy(int n) {
        Set<Integer> set = new HashSet<>();
        //Set去重，防止无限循环
        while(n != 1 && !set.contains(n)){
            set.add(n);
            n = getNextNumber(n);
        }
        return n==1;
    }

    //获取下一个数字的方法，res记录下一个数字，方法执行多次获取 res存一个数字
    //比如两位数，方法执行两次 res才是下一个数字
    public int getNextNumber(int n ){
        int res = 0;
        while(n != 0){
//            res = n%10 * n%10;
            res += (n % 10) * (n % 10);
            n /= 10;
        }
        return res;
    }
}
